Type: concept
Confidence: 0.95
Created: 2026-04-18
Updated: 2026-04-18
Tags: 机器学习深度学习序列建模

输入门(Input Gate)

概述 (50-200字符)

LSTM 三个门之一,决定将什么新信息写入细胞状态。由 iₜ(控制更新量,∈[0,1])和 C̃ₜ(候选新信息,∈[-1,1])组成,与遗忘门协作完成细胞状态的精确更新。

关键内容 (≥300字符, 用双链)

  1. 数学公式:iₜ = σ(Wi · [hₜ₋₁, xₜ] + bi),控制更新量 [0, 1];C̃ₜ = tanh(Wc · [hₜ₋₁, xₜ] + bc),候选新信息 [-1, 1]。两个子组件分别使用 Sigmoid激活函数 和 tanh 激活函数。
  2. 功能直觉输入门决定哪些新信息应该被写入 细胞状态(Cell State)。在语言模型中,遇到"小明"时输入门会写入主语信息;遇到"他"时保留"他=小明"的关联。iₜ 接近 1 时大量写入新信息,接近 0 时阻止写入。
  3. 细胞状态更新输入门遗忘门(Forget Gate) 协作:Cₜ = fₜ ⊙ Cₜ₋₁ + iₜ ⊙ C̃ₜ。遗忘门控制"擦除旧信息",输入门控制"写入新信息",两者共同实现细胞状态的精确调控。
  4. 候选值 C̃ₜ:使用 tanh 而非 sigmoid 生成候选值,输出范围 [-1, 1],允许细胞状态存储正负两种方向的信息,增强了表达能力。

来源

相关